package class5;

import common.TreeNode;
import common.TreeNodeUtils;

/**
 * https://leetcode.cn/problems/maximum-width-of-binary-tree/description/
 * 二叉树最大宽度
 * 解题思路：
 * 1.利用一个同步表，记录每个节点的编号，再计算编号的差值，就是该层的最大宽度
 * 左孩子：i*2
 * 右孩子：i*2+1
 */
public class Code21_widthOfBinaryTree {

    public static void main(String[] args) {
        TreeNode root = TreeNodeUtils.getBinarySearchTree();
        int ans = widthOfBinaryTree(root);
        System.out.println("ans = " + ans);
    }

    static int MAXN = 3001;
    static TreeNode[] queue = new TreeNode[MAXN];
    static int l, r;
    static int[] qct = new int[MAXN];

    public static int widthOfBinaryTree(TreeNode root) {
        int ans = 1;
        l = r = 0;
        queue[r] = root; // 节点内存地址
        qct[r++] = 1; // 节点编号
        while (l < r) {
            int size = r - l;// 当前队列的长度
            // qct[r-1]：当前队列最右的编号
            // qct[l]:最左位置编号
            // [r-1] - [l] + 1 ： r位置编号 - l位置编号 + 1
            ans = Math.max(ans, qct[r - 1] - qct[l] + 1);
            for (int i = 0; i < size; i++) {
                TreeNode cur = queue[l];
                int id = qct[l++];
                if (cur.left != null) {
                    queue[r] = cur.left;
                    qct[r++] = id * 2;
                }
                if (cur.right != null) {
                    queue[r] = cur.right;
                    qct[r++] = id * 2 + 1;
                }
            }
        }
        return ans;
    }

}
